package xyz.mumuwei.youthcourseserver.utils;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.concurrent.TimeUnit;

/**
 * @author mumuwei
 * @date 2022/2/19
 */
@Component
public class RedisUtil {

    private RedisTemplate<String, Object> redisTemplate;

    @Autowired
    public void setRedisTemplate(RedisTemplate<String, Object> redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    public RedisUtil(RedisTemplate<String, Object> redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    /**
     * 存储当前 key 以及 value值
     */
    public void set(String key, Object value){
        redisTemplate.opsForValue().set(key, value);
    }

    /**
     * 设置当前的key以及value值并且设置过期时间
     */
    public void set(String key, Object value, long timeout){
        redisTemplate.opsForValue().set(key, value, timeout, TimeUnit.MILLISECONDS);
    }

    /**
     * 设置当前的key以及value值并且设置过期时间
     */
    public void set(String key, Object value, long timeout, TimeUnit timeUnit){
        redisTemplate.opsForValue().set(key, value, timeout, timeUnit);
    }

    /**
     * 自增，控制访问
     */
    public Long increment(String key, long value){
        return redisTemplate.opsForValue().increment(key, value);
    }

    /**
     * 设置当前的key以及value值并且设置过期时间
     */
    public Object get(String key){
        return redisTemplate.opsForValue().get(key);
    }

    /**
     * 判断key是否存在
     */
    public boolean hasKey(String key){
            return Boolean.TRUE.equals(redisTemplate.hasKey(key));
    }

    /**
     * 判断key是否存在,若存在 返回 key的值
     */
    // public Object hasKey(String key){
    //     Boolean hasKey = redisTemplate.hasKey(key);
    //     if(Boolean.TRUE.equals(hasKey)){
    //         return redisTemplate.opsForValue().get(key);
    //     }
    //     return false;
    // }

    /**
     * 删除key
     */
    public void delete(String key){
        redisTemplate.delete(key);
    }

    /**
     * 指定 key的失效时间
     */
    public void expire(String key,long time){
        redisTemplate.expire(key, time, TimeUnit.MILLISECONDS);
    }

    /**
     * 指定 key的失效时间
     */
    public Boolean expireAt(String key, Date date) {
        return redisTemplate.expireAt(key, date);
    }

    /**
     * 根据key获取过期时间
     * @return key的过期时间
     */
    public long getExpire(String key){
        Long expire = redisTemplate.getExpire(key);
        return expire == null ? 0L : expire;
    }

}